package com.tianhaochen.springai.mongo;

import lombok.Getter;
import lombok.Setter;
import lombok.ToString;
import org.springframework.ai.chat.messages.AbstractMessage;
import org.springframework.data.annotation.Id;
import org.springframework.data.mongodb.core.mapping.Document;

import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;

@Getter
@Setter
@ToString
@Document("chat_record")
public class ChatRecord {
    @Id
    private String conversationId;

    private List<ChatMessage> messages = new ArrayList<>();

    private LocalDateTime createTime;
}
